package 中等.滑动窗口;

/**
 * 一位老师正在出一场由 n 道判断题构成的考试，每道题的答案为 true （用 'T' 表示）或者
 * false （用 'F' 表示）。老师想增加学生对自己做出答案的不确定性，方法是 最大化 有
 * 连续相同 结果的题数。（也就是连续出现 true 或者连续出现 false）。
 * <p>
 * 给你一个字符串 answerKey ，其中 answerKey[i] 是第 i 个问题的正确结果。除此以外，
 * 还给你一个整数 k ，表示你能进行以下操作的最多次数：
 * <p>
 * 每次操作中，将问题的正确答案改为 'T' 或者 'F' （也就是将 answerKey[i] 改为 'T' 或者 'F' ）。
 * 请你返回在不超过 k 次操作的情况下，最大 连续 'T' 或者 'F' 的数目。
 * <p>
 * 来源：力扣（LeetCode）
 * 链接：https://leetcode-cn.com/problems/maximize-the-confusion-of-an-exam
 */
public class 考试的最大困扰度_2024 {

    public static void main(String[] args) {

        String str = "TTFTTFTT";
        System.out.println(new 考试的最大困扰度_2024().maxConsecutiveAnswers2(str, 1));


    }

    /**
     * k 次操作的情况下，最大 连续 'T' 或者 'F' 的数目
     * 转换思路：
     * F和T较小的的数量不超过k的最长连续子数组
     *
     * @param answerKey
     * @param k
     * @return
     */
    public static int maxConsecutiveAnswers(String answerKey, int k) {
        int ans = 1, left = 0, right = 1;
        char[] answerKeys = answerKey.toCharArray();

        while (right < answerKey.length()) {
            int tCount = 0, fCount = 0;
            if (answerKeys[right] == 'F') {
                fCount++;
            } else {
                tCount++;
            }

            left = right - 1;
            while (left >= 0) {
                if (answerKeys[left] == 'F') {
                    fCount++;
                } else {
                    tCount++;
                }
                if (Math.min(fCount, tCount) > k) break;
                left--;
            }
            ans = Math.max(ans, left == -1 ? right + 1 : right - left);
            right++;
        }
        return ans;
    }

    /**
     * k 次操作的情况下，最大 连续 'T' 或者 'F' 的数目
     * 转换思路：
     * 最多 k 个 T 最长连续子数组
     * 最多 k 个 F 最长连续子数组
     *
     * @param answerKey
     * @param k
     * @return
     */
    public int maxConsecutiveAnswers2(String answerKey, int k) {
        return Math.max(getCharMaxLen(answerKey, 'F', k), getCharMaxLen(answerKey, 'T', k));
    }

    /**
     * 思路
     * 右边界逐渐+1，当碰到F时，curK++
     * 直到curK > K 时，左边界开始收缩，碰到F时，curK--
     * 使得窗口内的F数量始终不超过K
     *
     * @param k
     * @return
     */
    public int getCharMaxLen(String answerKey, char c, int k) {
        int maxLen = 1, curK = 0;
        for (int left = 0, right = 0; right < answerKey.length(); right++) {
            if (answerKey.charAt(right) == c) {
                curK++;
            }
            if (curK > k) {
                while (answerKey.charAt(left) != c) {  // 右边界一次最多增加一个元素，左边界也只要减少一个元素
                    left++;
                }
                left++;  // while循环是==c的时候停止，需要排除掉
                curK--;
            }
            maxLen = Math.max(maxLen, right - left + 1);
        }
        return maxLen;
    }

}
